#include "server.hpp"
#include <boost/bind.hpp>
#include <signal.h>

namespace http {
	namespace server {

		server::server(const std::string& address, const std::string& port,
			const std::string& doc_root)
			: io_context_(),
			signals_(io_context_),
			acceptor_(io_context_),
			connection_manager_(),
			new_connection_(),
			request_handler_(doc_root)
		{
			// Register to handle the signals that indicate when the server should exit.
			// It is safe to register for the same signal multiple times in a program,
			// provided all registration for the specified signal is made through Asio.
			signals_.add(SIGINT);
			signals_.add(SIGTERM);
#if defined(SIGQUIT)
			signals_.add(SIGQUIT);
#endif // defined(SIGQUIT)
			signals_.async_wait(boost::bind(&server::handle_stop, this));

			// Open the acceptor with the option to reuse the address (i.e. SO_REUSEADDR).
			boost::asio::ip::tcp::resolver resolver(io_context_);
			boost::asio::ip::tcp::endpoint endpoint =
				*resolver.resolve(address, port).begin();
			acceptor_.open(endpoint.protocol());
			acceptor_.set_option(boost::asio::ip::tcp::acceptor::reuse_address(true));
			acceptor_.bind(endpoint);
			acceptor_.listen();

			start_accept();
		}

		void server::run()
		{
			// The io_context::run() call will block until all asynchronous operations
			// have finished. While the server is running, there is always at least one
			// asynchronous operation outstanding: the asynchronous accept call waiting
			// for new incoming connections.
			io_context_.run();
		}

		void server::start_accept()
		{
			new_connection_.reset(new connection(io_context_,
				connection_manager_, request_handler_));
			acceptor_.async_accept(new_connection_->socket(),
				boost::bind(&server::handle_accept, this,
				boost::asio::placeholders::error));
		}

		void server::handle_accept(const boost::system::error_code& e)
		{
			// Check whether the server was stopped by a signal before this completion
			// handler had a chance to run.
			if (!acceptor_.is_open())
			{
				return;
			}

			if (!e)
			{
				connection_manager_.start(new_connection_);
			}

			start_accept();
		}

		void server::handle_stop()
		{
			// The server is stopped by cancelling all outstanding asynchronous
			// operations. Once all operations have finished the io_context::run() call
			// will exit.
			acceptor_.close();
			connection_manager_.stop_all();
		}

	} // namespace server
} // namespace http